import pandas as pd
import numpy as np

# 文件路径
input_file_path = r'D:\2024华为杯数学建模\2024C题-数据驱动与电磁损耗\C题\附件一数据拆解\材料4.csv'
output_file_path = r'D:\2024华为杯数学建模\2024C题-数据驱动与电磁损耗\C题\附件一数据拆解\Q2\Q2磁芯材料4正弦波形.csv'

# 读取CSV文件，指定编码（如需要，请根据实际情况调整编码）
data = pd.read_csv(input_file_path, encoding='latin1')  # 尝试不同的编码格式

# 提取前四列
extracted_data = data.iloc[:, :4]

# 筛选出第四列值为1的数据
filtered_data = extracted_data[extracted_data.iloc[:, 3] == 1]

# 计算第5至1029列的峰值
# 假设数据从第5列到第1029列是数值列
peak_values = data.iloc[:, 4:1029].max(axis=1)

# 将峰值添加到筛选后的数据中
filtered_data['磁通密度峰值'] = peak_values[filtered_data.index]

# 保存到新的CSV文件
filtered_data.to_csv(output_file_path, index=False)

print("数据提取完成，已保存至:", output_file_path)
